package main

import (
	"fmt"
	"net"
	"sync"
)

type Server struct {
	Ip string
	Port int

	// 在线用户列表
	OnlineMap map[string]*User
	mapLock sync.RWMutex

	// 消息广播的channel
	Message chan string
}

// 创建一个server的接口
func NewServer(ip string, port int) *Server {
	server := &Server {
		Ip: ip,
		Port: port,
		OnlineMap: make(map[string]*User),
		Message: make(chan string),
	}
	
	return server
}

func (this *Server) Handler(conn net.Conn) {
	// fmt.Println("链接建立成功！")

	user := NewUser(conn)
	// 用户上线了， 将用户加入到OnlineMap表中
	this.mapLock.Lock()
	this.OnlineMap[user.Name] = user
	this.mapLock.Unlock()

	// 广播上线的消息
	this.BroadCast(user, "用户已上线")

	// 当前handler 阻塞
	select {}


}
// 监听message 广播消息channel 的goroutine， 一旦有消息就发送全部的在线用户
func (this *Server) ListenMessage () {
	for {
		msg := <-this.Message

		// 将message 发送给在线所有的用户
		this.mapLock.Lock()
		for _,cli := range this.OnlineMap {
			cli.C <- msg
		}
		this.mapLock.Unlock()
	}
}


func (this *Server) BroadCast(user *User, msg string) {
	sendMsg := "[" + user.Addr + "]" + user.Name + ":" + msg
	this.Message <-sendMsg
}

func (this *Server) Start() {
	// socket listen
	listener, err := net.Listen("tcp", fmt.Sprintf("%s:%d", this.Ip, this.Port))
	if err != nil {
		fmt.Println("net.Listen err:", err)
		return
	}
	// close listen socket
	defer listener.Close()

	// 启动监听Message 的routine
	go this.ListenMessage()
	for {
		// accept
		// fmt.Println("建立前...")
		conn, err := listener.Accept()
		// fmt.Println("建立后...")
		if err != nil {
			fmt.Println("listener accept err:", err)
			continue
		}
		// do handler
		go this.Handler(conn)
	}
	
}